import torchvision
import torch
import torch.nn as nn
from torchsummary import summary

model = torchvision.models.vgg16(pretrained=True)
# summary(model,torch.zeros(1,3,224,224))
print(model)